package com.shuai.baibi.service.impl;

import com.shuai.baibi.mapper.AITemplateMapper;
import com.shuai.baibi.model.entity.AITemplate;
import com.shuai.baibi.service.AITemplateService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class AITemplateServiceImpl implements AITemplateService {
    @Autowired
    private AITemplateMapper aiTemplateMapper;

    @Override
    public AITemplate getPromptByModelName(String modelName) {
        return aiTemplateMapper.getAITemplateByModelName(modelName);
    }
}
